﻿/**
 * 通用js方法封装处理
 * Copyright (c) 2019 ruoyi
 */
import dayjs from "dayjs";

const baseURL = process.env.VUE_APP_BASE_API;

export function dateFormatter(row, column) {
    return dayjs(row.createdAt).format("YYYY-MM-DD HH:mm:ss");
}

export function formatDate(date, fmt = "yyyy-MM-dd") {
    let opt = {
        "y+": date.getFullYear().toString(),
        "M+": (date.getMonth() + 1).toString(),
        "d+": date.getDate().toString(),
        "h+": date.getHours().toString(),
        "m+": date.getMinutes().toString(),
        "s+": date.getSeconds().toString(),
    };
    let ret = null;
    for (let k in opt) {
        ret = new RegExp("(" + k + ")").exec(fmt);
        if (ret) {
            fmt = fmt.replace(
                ret[1],
                ret[1].length == 1
                    ? opt[k]
                    : opt[k].padStart(ret[1].length, "0")
            );
        }
    }
    return fmt;
}

// 日期格式化
export function parseTime(time, pattern) {
    if (arguments.length === 0 || !time) {
        return null;
    }
    const format = pattern || "{y}-{m}-{d} {h}:{i}:{s}";
    let date;
    if (typeof time === "object") {
        date = time;
    } else {
        if (typeof time === "string" && /^[0-9]+$/.test(time)) {
            time = parseInt(time);
        } else if (typeof time === "string") {
            time = time.replace(new RegExp(/-/gm), "/");
        }
        if (typeof time === "number" && time.toString().length === 10) {
            time = time * 1000;
        }
        date = new Date(time);
    }
    const formatObj = {
        y: date.getFullYear(),
        m: date.getMonth() + 1,
        d: date.getDate(),
        h: date.getHours(),
        i: date.getMinutes(),
        s: date.getSeconds(),
        a: date.getDay(),
    };
    const time_str = format.replace(/{(y|m|d|h|i|s|a)+}/g, (result, key) => {
        let value = formatObj[key];
        // Note: getDay() returns 0 on Sunday
        if (key === "a") {
            return ["日", "一", "二", "三", "四", "五", "六"][value];
        }
        if (result.length > 0 && value < 10) {
            value = "0" + value;
        }
        return value || 0;
    });
    return time_str;
}

// 表单重置
export function resetForm(refName) {
    if (this.$refs[refName]) {
        this.$refs[refName].resetFields();
    }
}

// 添加日期范围
export function addDateRange(params, dateRange, propName) {
    var search = params;
    search.params = {};
    if (null != dateRange && "" != dateRange) {
        if (typeof propName === "undefined") {
            search.params["beginTime"] = dateRange[0];
            search.params["endTime"] = dateRange[1];
        } else {
            search.params["begin" + propName] = dateRange[0];
            search.params["end" + propName] = dateRange[1];
        }
    }
    return search;
}

// 添加日期范围（舆情大屏用）
export function newAddDateRange(params, dateRange, propName, payload) {
    const search = params;
    search.params = {};
    if (dateRange != null && dateRange != "") {
        if (typeof propName === "undefined") {
            search.params.beginTime = dateRange[0];
            search.params.endTime = dateRange[1];
        } else if (propName == "custom") {
            search[payload.keyname] = dateRange.join(",");
        } else if (propName == "custom_be") {
            search[payload.begin] = dateRange[0];
            search[payload.end] = dateRange[1];
        } else {
            search.params[`begin${propName}`] = dateRange[0];
            search.params[`end${propName}`] = dateRange[1];
        }
    } else if (!dateRange || dateRange?.length == 0) {
        if (propName == "custom") {
            search[payload.keyname] = "";
        } else if (propName == "custom_be") {
            search[payload.begin] = "";
            search[payload.end] = "";
        }
    }
    return search;
}

// 回显数据字典
export function selectDictLabel(datas, value) {
    var actions = [];
    Object.keys(datas).some((key) => {
        if (datas[key].dictValue == "" + value) {
            actions.push(datas[key].dictLabel);
            return true;
        }
    });
    return actions.join("");
}

// 回显数据字典（字符串数组）
export function selectDictLabels(datas, value, separator) {
    var actions = [];
    var currentSeparator = undefined === separator ? "," : separator;
    var temp = value.split(currentSeparator);
    Object.keys(value.split(currentSeparator)).some((val) => {
        Object.keys(datas).some((key) => {
            if (datas[key].dictValue == "" + temp[val]) {
                actions.push(datas[key].dictLabel + currentSeparator);
            }
        });
    });
    return actions.join("").substring(0, actions.join("").length - 1);
}

// 通用下载方法
export function download(fileName) {
    window.location.href =
        baseURL +
        "/common/download?fileName=" +
        encodeURI(fileName) +
        "&delete=" +
        true;
}

// 字符串格式化(%s )
export function sprintf(str) {
    var args = arguments,
        flag = true,
        i = 1;
    str = str.replace(/%s/g, function () {
        var arg = args[i++];
        if (typeof arg === "undefined") {
            flag = false;
            return "";
        }
        return arg;
    });
    return flag ? str : "";
}

// 转换字符串，undefined,null等转化为""
export function praseStrEmpty(str) {
    if (!str || str == "undefined" || str == "null") {
        return "";
    }
    return str;
}

/**
 *
 * @param data 数据源
 * @param id字段 默认 'id'
 * @param parentId 父节点字段 默认 'parentId'
 * @param children 孩子节点字段 默认 'children'
 * @param rootId 根Id 默认 0
 * @returns {{obj, list: Array}}
 */
export function handleTree1(
    data,
    id = "id",
    parentId = "parentId",
    children = "children",
    rootId = 0
) {
    const obj = {};
    data.forEach((item) => {
        item[children] = [];
        obj[item[id]] = item;
    });
    const tree = [];
    data.forEach((list) => {
        if (list[parentId] !== rootId) {
            if (obj[list[parentId]]) {
                obj[list[parentId]][children].push(list);
            }
        } else {
            tree.push(list);
        }
    });
    return {
        tree,
        obj,
    };
}

/**
 * 构造树型结构数据
 * @param {*} data 数据源
 * @param {*} id id字段 默认 'id'
 * @param {*} parentId 父节点字段 默认 'parentId'
 * @param {*} children 孩子节点字段 默认 'children'
 */
export function handleTree(data, id, parentId, children) {
    let config = {
        id: id || "id",
        parentId: parentId || "parentId",
        childrenList: children || "children",
    };

    var childrenListMap = {};
    var nodeIds = {};
    var tree = [];

    for (let d of data) {
        let parentId = d[config.parentId];
        if (childrenListMap[parentId] == null) {
            childrenListMap[parentId] = [];
        }
        nodeIds[d[config.id]] = d;
        childrenListMap[parentId].push(d);
    }

    for (let d of data) {
        let parentId = d[config.parentId];
        if (nodeIds[parentId] == null) {
            tree.push(d);
        }
    }

    for (let t of tree) {
        adaptToChildrenList(t);
    }

    function adaptToChildrenList(o) {
        if (childrenListMap[o[config.id]] !== null) {
            o[config.childrenList] = childrenListMap[o[config.id]];
        }
        if (o[config.childrenList]) {
            for (let c of o[config.childrenList]) {
                adaptToChildrenList(c);
            }
        }
    }

    return tree;
}

/**
 * 递归查找，获取children
 */
const getChildren = (data, result, pid, parentId) => {
    for (let item of data) {
        if (item[parentId] === pid) {
            const newItem = { ...item, children: [] };
            result.push(newItem);
            getChildren(data, newItem.children, item.id, parentId);
        }
    }
};

/**
 * 转换方法
 * let arr = [
 {id: 1, name: '部门1', pid: 0},
 {id: 2, name: '部门2', pid: 1},
 {id: 3, name: '部门3', pid: 1},
 {id: 4, name: '部门4', pid: 3},
 {id: 5, name: '部门5', pid: 4},
 ]
 */
export function arrayToTree(data = [], pid = 0, parentId = "pid") {
    const result = [];
    getChildren(data, result, pid, parentId);
    return result;
}

/**
 * 转换方法
 * let arr = [
 {id: 1, name: '部门1', pid: 0},
 {id: 2, name: '部门2', pid: 1},
 {id: 3, name: '部门3', pid: 1},
 {id: 4, name: '部门4', pid: 3},
 {id: 5, name: '部门5', pid: 4},
 ]
 */
export function arrayToTree2(items = [], cid = "id", parentId = "pid") {
    const result = []; // 存放结果集
    const itemMap = {}; //
    for (const item of items) {
        const id = item[cid];
        const pid = item[parentId];
        if (!itemMap[id]) {
            itemMap[id] = { children: [] };
        }
        itemMap[id] = { ...item, children: itemMap[id]["children"] };
        const treeItem = itemMap[id];
        if (pid === 0) {
            result.push(treeItem);
        } else {
            if (!itemMap[pid]) {
                itemMap[pid] = { children: [] };
            }
            itemMap[pid].children.push(treeItem);
        }
    }
    return result;
}
